Use list of derived class as list of base class in Python